package P621_任务调度器;

import java.util.*;

/**
 * 给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。其中每个字母表示一种不同种类的任务。任务可以以任意顺序执行，并且每个任务都可以在 1 个单位时间内执行完。在任何一个单位时间，CPU 可以完成一个任务，或者处于待命状态。
 * <p>
 * 然而，两个 相同种类 的任务之间必须有长度为整数 n 的冷却时间，因此至少有连续 n 个单位时间内 CPU 在执行不同的任务，或者在待命状态。
 * <p>
 * 你需要计算完成所有任务所需要的 最短时间 。
 * <p>
 *  
 * <p>
 * 示例 1：
 * <p>
 * 输入：tasks = ['A','A','A','B','B','B'], n = 2
 * 输出：8
 * 解释：A -> B -> (待命) -> A -> B -> (待命) -> A -> B
 * 在本示例中，两个相同类型任务之间必须间隔长度为 n = 2 的冷却时间，而执行一个任务只需要一个单位时间，所以中间出现了（待命）状态。
 * 示例 2：
 * <p>
 * 输入：tasks = ['A','A','A','B','B','B'], n = 0
 * 输出：6
 * 解释：在这种情况下，任何大小为 6 的排列都可以满足要求，因为 n = 0
 * ['A','A','A','B','B','B']
 * ['A','B','A','B','A','B']
 * ['B','B','B','A','A','A']
 * ...
 * 诸如此类
 * 示例 3：
 * <p>
 * 输入：tasks = ['A','A','A','A','A','A','B','C','D','E','F','G'], n = 2
 * 输出：16
 * 解释：一种可能的解决方案是：
 * A -> B -> C -> A -> D -> E -> A -> F -> G -> A -> (待命) -> (待命) -> A -> (待命) -> (待命) -> A
 *  
 * <p>
 * 提示：
 * <p>
 * 1 <= task.length <= 104
 * tasks[i] 是大写英文字母
 * n 的取值范围为 [0, 100]
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/task-scheduler
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Solution {
    //应该是贪心算法，每次尽量拿多的,不行就拿次多的，完全不行就休息
    //维持最大堆，堆顶n个数
    public int leastInterval(char[] tasks, int n) {
        if (n == 0) {
            return tasks.length;
        }

        HashMap<Character, Integer> map = new HashMap<>();

        PriorityQueue<Character> heap = new PriorityQueue<>(n + 1, (o1, o2) -> map.getOrDefault(o2, 0).compareTo(map.getOrDefault(o1, 0)));
        for (char c : tasks) {
            map.put(c, map.getOrDefault(c, 0) + 1);
        }
        for (char c : map.keySet()) {
            heap.offer(c);
        }

        int count = 0;
        int tmp = 0;
        int inner;
        Character c;
        while(heap.size() > 0 ) {
            inner = 0;
            while ((c = heap.poll()) != null && (inner++ < n + 1)) {
                count ++;
                tmp = map.get(c);
                if (tmp <= 1) {
                    map.remove(c);
                } else {
                    map.put(c, tmp - 1);
                }
            }
            //没有任务了
            if(map.size() == 0){
                return count;
            }
            if(inner < n + 1){
                //不超过n + 1证明一次不够用了，需要填充待命时间
                count += n + 1 - inner;

            }
            heap.clear();
            for (char ch : map.keySet()) {
                heap.offer(ch);
            }
        }

        return count;
    }

    public static void main(String args[]){
        System.out.println(new Solution().leastInterval(new char[]{'A','A','A','B','B','B'}, 2));
    }
}
